"""
重置数据库脚本
"""
import os
import sys
from pathlib import Path

# 添加项目根目录到Python路径
sys.path.insert(0, str(Path(__file__).parent))

from lemon_backend.models.db import init_db, engine, Base

def reset_database():
    """删除所有表并重新创建"""
    print("正在重置数据库...")

    # 删除所有表
    Base.metadata.drop_all(engine)
    print("已删除所有表")

    # 重新创建所有表
    init_db()
    print("已重新创建所有表")

    print("数据库重置完成")

if __name__ == "__main__":
    # 确认操作
    confirm = input("此操作将删除所有数据，是否继续？(y/n): ")
    if confirm.lower() == 'y':
        reset_database()
    else:
        print("操作已取消")
